"""
折线图开发前，对txt中的json数据进行处理，将数据转换为折线图所需的数据格式。
"""
import json

# 处理数据
f_us = open("./美国.txt", "r", encoding="utf-8")
us_data = f_us.read()

f_jp = open("./日本.txt", "r", encoding="utf-8")
jp_data = f_jp.read()

f_in = open("./印度.txt", "r", encoding="utf-8")
in_data = f_in.read()

# 去掉不合法JSON规范的开头字符
us_data = us_data.replace('jsonp_1629344292311_69436(', '')
jp_data = jp_data.replace('jsonp_1629350871167_29498(', '')
in_data = in_data.replace('jsonp_1629350745930_63180(', '')

# 去掉不合法JSON规范的结尾字符
us_data = us_data[:-2]
jp_data = jp_data[:-2]
in_data = in_data[:-2]

# JSON转Python字典
us_dict = json.loads(us_data)
jp_dict = json.loads(jp_data)
in_dict = json.loads(in_data)

# 获取trend key
us_trend_key_data = us_dict['data'][0]['trend']
jp_trend_key_data = jp_dict['data'][0]['trend']
in_trend_key_data = in_dict['data'][0]['trend']

# 换取日期数据，用于x轴，取2020年（到314下标结束）
us_x_data = us_trend_key_data['updateDate'][:314]
jp_x_data = jp_trend_key_data['updateDate'][:314]
in_x_data = in_trend_key_data['updateDate'][:314]

# 获吹确认数据，用于y轴，收2020年（到314下标结束）
us_y_data = us_trend_key_data['list'][0]['data'][:314]
jp_y_data = jp_trend_key_data['list'][0]['data'][:314]
in_y_data = in_trend_key_data['list'][0]['data'][:314]
